from sqlalchemy import Column, Integer, String
from database.db_init import Base


class Ai_model(Base):
    __tablename__ = "AI_Model"
    ID = Column(Integer, primary_key=True, index=True)
    ModelName = Column(String, unique=False)
    ModelFunction = Column(String, unique=False)
    InputType = Column(String, unique=False)
    OutputType = Column(String, unique=False)
    ModelAPIAddress = Column(String, unique=False)
